from .storeModel import StoreModel


class StoreUser(StoreModel):
    def __init__(self, name, Model):
        super().__init__(name, Model)
        self.sid = None
        self.session = None
        self.allowed_company_id = None

    def login(self, user, password):
        # print('in, store user login')
        # print(user, password)
        # 登录,  session
        # 获取 名下的所有公司
        # 前端 有一个地方 选择公司.
        # 可以简化处理, erp 用户 默认都只有一个公司
        # 取当前公司 做为 allowed_company_id
        # 后续所有的 call , 需要加上 centext: allowed_company_id

        self.api.login(user, password)
        # print('---1 api----')
        # print(self.api.version)
        # print('----2 env---')
        # print(self.env.session)

        # self.sid = self.env.session['session_id']
        self.sid = self.env.sid
        self.session = self.env.session

        self.allowed_company_id = self.env.allowed_company_id

    def setCompany(self, value):
        self.env.allowed_company_id = value
        # 这个值 要写到 localstore 里.
        # 刷新页面时,
        # 读 sessionid, 重新 从服务器取 session info,
        # 且 从 localstore 取 allowed_company_id, 写到 api 里
